tensor([[0.6000, 0.6000, 0.6000, 0.6000, 0.6000],
        [0.4000, 0.4000, 0.4000, 0.4000, 0.4000]], grad_fn=<RoundBackward1>)